package com.youzikj.wx.comm.util;

import java.io.Writer;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Pattern;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;

import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.core.util.QuickWriter;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import com.thoughtworks.xstream.io.xml.PrettyPrintWriter;
import com.thoughtworks.xstream.io.xml.XppDriver;
import com.youzikj.wx.comm.respmsg.Article;
import com.youzikj.wx.comm.respmsg.NewsMessage;
import com.youzikj.wx.comm.respmsg.RespImageMessage;
import com.youzikj.wx.comm.respmsg.RespMusicMessage;
import com.youzikj.wx.comm.respmsg.RespTextMessage;
import com.youzikj.wx.comm.respmsg.RespVideoMessage;
import com.youzikj.wx.comm.respmsg.RespVoiceMessage;

/**
 * 消息处理工具类
 * 
 * @author Administrator
 *
 */
public class MessageUtil {
	// 请求消息类型：文本
	public static final String REQ_MESSAGE_TYPE_TEXT = "text";
	// 请求消息类型：图片
	public static final String REQ_MESSAGE_TYPE_IMAGE = "image";
	// 请求消息类型：语音
	public static final String REQ_MESSAGE_TYPE_VOICE = "voice";
	// 请求消息类型：视频
	public static final String REQ_MESSAGE_TYPE_VIDEO = "video";
	// 请求消息类型：地理位置
	public static final String REQ_MESSAGE_TYPE_LOCATION = "location";
	// 请求消息类型：链接
	public static final String REQ_MESSAGE_TYPE_LINK = "link";
	// 请求消息类型：事件推送
	public static final String REQ_MESSAGE_TYPE_EVENT = "event";

	// 事件消息类型：订阅
	public static final String EVENT_MESSAGE_TYPE_SUBSCRIBE = "subscribe";
	// 事件消息类型：取消订阅
	public static final String EVENT_MESSAGE_TYPE_UNSUBSCRIBE = "unsubscribe";
	// 事件消息类型：关注扫描带参数二维码
	public static final String EVENT_MESSAGE_TYPE_SCAN = "scan";
	// 事件消息类型：上报地理位置
	public static final String EVENT_MESSAGE_TYPE_LOCATION = "location";
	// 事件消息类型：自定义菜单
	public static final String EVENT_MESSAGE_TYPE_CLICK = "click";

	// 响应消息类型:文本
	public static final String RESP_MESSAGE_TYPE_TEXT = "text";
	// 响应消息类型:图片
	public static final String RESP_MESSAGE_TYPE_IMAGE = "image";
	// 响应消息类型:语音
	public static final String RESP_MESSAGE_TYPE_VOICE = "voice";
	// 响应消息类型:视频
	public static final String RESP_MESSAGE_TYPE_VIDEO = "video";
	// 响应消息类型:音乐
	public static final String RESP_MESSAGE_TYPE_MUSIC = "music";
	// 响应消息类型:图文
	public static final String RESP_MESSAGE_TYPE_NEWS = "news";

	/**
	 * 解析微信发来的请求(xml)
	 * 
	 * @param request
	 * @return
	 */
	public static Map<String, String> parseXml(HttpServletRequest request)
			throws Exception {
		Map<String, String> map = new HashMap<String, String>();
		// 从reqeust中获取输入流
		ServletInputStream inputStream = request.getInputStream();
		// 读取输入流
		SAXReader reader = new SAXReader();
		Document document = reader.read(inputStream);
		Element root = document.getRootElement();
		List<Element> list = root.elements();

		// 遍历所有子节点
		for (Element e : list) {
			map.put(e.getName(), e.getText());
		}
		inputStream.close();
		inputStream = null;
		return map;
	}

	/**
	 * 扩展xstream使其支持CData
	 */
	private static XStream xstream = new XStream(new XppDriver() {
		public HierarchicalStreamWriter createWriter(Writer out) {
			return new PrettyPrintWriter(out) {
				//对xml节点的转换增加CDATA标记， true增加
				boolean cdata = false;

				public void startNode(String name, Class clazz) {
					if(!name.equals("xml")){
						char[] arr = name.toCharArray();
						if(arr[0]>='a' && arr[0]<='z'){
							 //ASCII码，大写字母和小写字符之间数值上差32
			                arr[0] = (char) ((int) arr[0] - 32);
						}
						  name = new String(arr);
					}
					super.startNode(name, clazz);
				}
				
				   @Override
	                public void setValue(String text) {
	                    
	                    if(text!=null && !"".equals(text)){
	                        //浮点型判断
	                        Pattern patternInt = Pattern.compile("[0-9]*(\\.?)[0-9]*");
	                        //整型判断
	                        Pattern patternFloat = Pattern.compile("[0-9]+");
	                        //如果是整数或浮点数 就不要加[CDATA[]了
	                        if(patternInt.matcher(text).matches() || patternFloat.matcher(text).matches()){
	                        cdata = false;
	                        }else{
	                        cdata = true;
	                        }
	                    }
	                    super.setValue(text);  
	                }

				protected void writeText(QuickWriter writer, String text) {
					if (cdata) {
						writer.write("<![CDATA[");
						writer.write(text);
						writer.write("]]>");
					} else {
						writer.write(text);
					}
				}
			};
		}
	});

	/**
	 * 文本消息对象转换成XML
	 * 
	 * @param textMessage
	 * @return
	 */
	public static String messageToXml(RespTextMessage textMessage) {
		xstream.alias("xml", textMessage.getClass());
		return xstream.toXML(textMessage);
	}

	/**
	 * 图片消息对象转换成xml
	 * 
	 * @param imageMessage
	 * @return
	 */
	public static String messageToXml(RespImageMessage imageMessage) {
		xstream.alias("xml", imageMessage.getClass());
		return xstream.toXML(imageMessage);
	}

	/**
	 * 语音消息对象转换成xml
	 * 
	 * @param voiceMessage
	 * @return
	 */
	public static String messageToXml(RespVoiceMessage voiceMessage) {
		xstream.alias("xml", voiceMessage.getClass());
		return xstream.toXML(voiceMessage);
	}

	/**
	 * 视频消息对象转换成XML
	 * 
	 * @param videoMessage
	 * @return
	 */
	public static String messageToXml(RespVideoMessage videoMessage) {
		xstream.alias("xml", videoMessage.getClass());
		return xstream.toXML(videoMessage);
	}

	/**
	 * 音乐消息对象转换成xml
	 * 
	 * @param musicMessage
	 * @return
	 */
	public static String messageToXml(RespMusicMessage musicMessage) {
		xstream.alias("xml", musicMessage.getClass());
		return xstream.toXML(musicMessage);
	}
	
	/**
	 * 图文消息对象转换成xml
	 * @param newsMessage
	 * @return
	 */
	public static String messageToXml(NewsMessage newsMessage) {
		xstream.alias("xml", newsMessage.getClass());
		xstream.alias("item", new Article().getClass());
		return xstream.toXML(newsMessage);
	}

}
